package com.yeung.swordoffer;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;

/**
 * 题目描述
 * 输入一颗二叉树的根节点和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。
 * 路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
 * (注意: 在返回值的list中，数组长度大的数组靠前)
 */
public class 二叉树中和为某一值的路径 {
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
        ArrayList<ArrayList<Integer>> list = new ArrayList<>();
        getPath(root, target, 0, list, new ArrayList<>());
        Collections.sort(list, new Comparator<ArrayList<Integer>>() {
            @Override
            public int compare(ArrayList<Integer> o1, ArrayList<Integer> o2) {
                return -(o1.size() - o2.size());
            }
        });
        return list;
    }

    static void getPath(TreeNode root, int target, int nowSum, ArrayList<ArrayList<Integer>> all, ArrayList<Integer> tmp) {
        //不满足条件, 且无后续节点
        if (root == null) return;
        //满足条件
        if (nowSum + root.val == target && root.left == null && root.right == null) {
            tmp.add(root.val);
            all.add(tmp);
            return;//满足条件 ,返回
        }
        //继续向下递归
        tmp.add(root.val);
        getPath(root.left, target, nowSum + root.val, all, (ArrayList<Integer>) tmp.clone());
        getPath(root.right, target, nowSum + root.val, all, (ArrayList<Integer>) tmp.clone());
    }
}
